#!/usr/bin/env python3

# Following two lines needed so all variables are shown when printing the Dataset
import xarray as xr
import xhermes

xr.set_options(display_max_rows=1000)

# Set better figure size
from matplotlib import pyplot as plt

plt.rcParams["figure.figsize"] = (16,8)

ds = xhermes.open(".", geometry="toroidal",
    gridfilepath="bout.grd.nc",
    keep_yboundaries=False,
    keep_xboundaries=False)

# Get rid of size-1 toroidal direction
ds = ds.squeeze()

# Make an animation
# Note: saving a gif can be slow. Comment out `save_as` argument to disable.
ds.bout.animate_polygon_list(
    ["Nd","Pd"],
    #["Nd"],
    poloidal_plot=True,
    show=False,
    save_as="hermes_animation_new",
)